package bintree.map_set.LeetCode;

import java.util.HashMap;
import java.util.Map;
import java.util.TreeMap;

//138. 复制带随机指针的链表
public class Num138_copyRandomList {
    public Node copyRandomList(Node head) {
        //1. 遍历原链表，构造Map集合维护原链表节点和新链表节点的映射关系
        //原1 = 新1
        Map<Node, Node> map = new HashMap<>();
        for(Node x = head; x != null; x = x.next){
            Node node = new Node(x.val);
            map.put(x, node);//原1 = 新1 两者建立关系
        }
        // 2. 再次遍历原链表，通过映射关系维护新链表的next和 random
        for(Node x = head; x != null; x = x.next){
            //原1.next = 原3
            //新1.next = 新3
            //get是输入key返回对应的vale，相当于输入原地址，返回新地址
            map.get(x).next = map.get(x.next);
            //原1.random = 原3
            //新1.random = 新3
            map.get(x).random = map.get(x.random);
        }
        return map.get(head);//因为原链表和新链表是映射关系，输入原地址，返回新地址；
    }
}

class Node {
    int val;
    Node next;
    Node random;

    public Node(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}
